from EduData import get_data
import os
import shutil
from pathlib import Path

DATASET = Path(__file__).parent.parent / "dataset"

if not DATASET.exists():
    os.makedirs(DATASET)

# get_data("cdbd-a0910", DATASET)
# get_data("junyi", DATASET)
get_data("assistment-2017", DATASET)
# get_data("NIPS-2020", DATASET)

# shutil.move(DATASET / "anonymized_full_release_competition_dataset", DATASET / "assistment-2017")
shutil.move(
    DATASET / "assistment-2017" / "anonymized_full_release_competition_dataset.csv",
    DATASET / "assistment-2017" / "data.csv",
)

# os.remove(DATASET / "anonymized_full_release_competition_dataset.zip")
# os.remove(DATASET / "junyi.rar")

# os.remove(DATASET / "NIPS2020" / "public_data.zip")

# shutil.move(DATASET / "NIPS2020" / "public_data", DATASET / "NIPS2020-data")
# shutil.rmtree(DATASET / "NIPS2020")
# shutil.move(DATASET / "NIPS2020-data", DATASET / "NIPS2020")
